/*
 * Licensed to the Tai Ping Jin Ke
 *
 * Copyright (c) 2022 .
 * All rights reserved.
 * 项目名称：太平金科-通用能力封装层-mybatis
 * 版权说明：本软件属太平金融科技服务(上海)有限公司所有，在未获得太平金融科技服务(上海)有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.taipingframework.mybatis.unid.plus;

import cn.hutool.core.util.IdUtil;
import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
import com.taipingframework.utility.unid.AppGenKey;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.lang.NonNull;
import org.springframework.stereotype.Component;

/**
 * mybatis-plus 分布式全局唯一ID生成策略
 */
@Component
public final class DistributedGlobalIdGenerator implements IdentifierGenerator {

    private static AppGenKey genKey;

    /**
     * 设置 创建分布式全局ID 的对象
     */
    @Autowired
    public void setGenKey(@NonNull AppGenKey genKey) {
        DistributedGlobalIdGenerator.setSyncGenKey(genKey);
    }

    private static synchronized void setSyncGenKey(AppGenKey genKey) {
        DistributedGlobalIdGenerator.genKey = genKey;
    }

    @Override
    public Number nextId(Object entity) {
        return genKey.getUniqueKey();
    }

    @Override
    public String nextUUID(Object entity) {
        return IdUtil.simpleUUID();
    }
}
